input_dim = 2048
num_hiddens = 1024
num_head = 1
dropout = 0.3
window = 30
lr = 1e-4
epochs = 1000
batch_size = 16
bearing_condition = 1